Desbloqueando o Poder do Seu Django Admin: Ações Personalizadas Explicadas

            # myapp/models.py
from django.db import models

class Product(models.Model):
    name = models.CharField(max_length=200)
    price = models.DecimalField(max_digits=10, decimal_places=2)
    is_discounted = models.BooleanField(default=False)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.name

            

              
                Copy
              
            
          

            # myapp/admin.py
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest
from .models import Product

@admin.register(Product)
class ProductAdmin(admin.ModelAdmin):
    list_display = ('name', 'price', 'is_discounted', 'created_at')
    list_filter = ('is_discounted', 'created_at')
    search_fields = ('name',)

    # Definir a função da ação personalizada do admin
    def make_discounted(self, request: HttpRequest, queryset: QuerySet):
        updated_count = queryset.update(is_discounted=True)
        self.message_user(
            request,
            f"{updated_count} produto(s) foram marcados como com desconto com sucesso.",
            messages.SUCCESS
        )
    make_discounted.short_description = "Marcar produtos selecionados como com desconto"

    # Registrar a ação com o ModelAdmin
    actions = [make_discounted]

            

              
                Copy
              
            
          

            # myapp/models.py (assumindo um modelo Post)
from django.db import models

class Post(models.Model):
    title = models.CharField(max_length=255)
    content = models.TextField()
    status = models.CharField(max_length=20, default='draft', choices=[
        ('draft', 'Draft'),
        ('pending', 'Pending Review'),
        ('approved', 'Approved'),
        ('rejected', 'Rejected'),
    ])
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.title

            

              
                Copy
              
            
          

            # myapp/forms.py
from django import forms

class ConfirmationForm(forms.Form):
    confirm = forms.BooleanField(
        label="Tem certeza de que deseja realizar esta ação?",
        required=True,
        widget=forms.HiddenInput # Lideraremos a exibição no template
    )
    _selected_action = forms.CharField(widget=forms.HiddenInput)
    action = forms.CharField(widget=forms.HiddenInput)

            

              
                Copy
              
            
          

            # myapp/admin.py
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest, HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from .models import Post
from .forms import ConfirmationForm

@admin.register(Post)
class PostAdmin(admin.ModelAdmin):
    list_display = ('title', 'status', 'created_at')
    list_filter = ('status',)
    search_fields = ('title',)

    def mark_posts_approved(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        # Verificar se o usuário confirmou a ação
        if 'apply' in request.POST:
            form = ConfirmationForm(request.POST)
            if form.is_valid():
                updated_count = queryset.update(status='approved')
                self.message_user(
                    request,
                    f"{updated_count} post(s) foram marcados como aprovados com sucesso.",
                    messages.SUCCESS
                )
                return HttpResponseRedirect(request.get_full_path())
        
        # Se não confirmada, ou requisição GET, mostrar página de confirmação
        else:
            # Armazenar as chaves primárias dos objetos selecionados em um campo oculto
            # Isso é crucial para passar a seleção pela página de confirmação
            context = self.admin_site.each_context(request)
            context['queryset'] = queryset
            context['form'] = ConfirmationForm(initial={
                '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
                'action': 'mark_posts_approved',
            })
            context['action_name'] = self.mark_posts_approved.short_description
            context['title'] = _("Confirmar Ação")

            # Renderizar um template de confirmação personalizado
            return render(request, 'admin/confirmation_action.html', context)

    mark_posts_approved.short_description = _("Marcar posts selecionados como aprovados")

    actions = [mark_posts_approved]

            

              
                Copy
              
            
          

            {# templates/admin/confirmation_action.html #}
{% extends "admin/base_site.html" %}
{% load i18n admin_urls static admin_modify %}

{% block extrastyle %}{{ block.super }}
    
{% endblock %}

{% block content %}
    

        

            {% csrf_token %}
            
{{ action_name }}
            
Você está prestes a executar a ação "{{ action_name }}" nos seguintes {{ queryset.count }} objeto(s):
            

                {% for obj in queryset %}
                    
{{ obj }}
                {% endfor %}
            
            
Você tem certeza absoluta de que deseja prosseguir?

            {{ form.as_p }}

            
            {% for item in form._selected_action.value %}
                
            {% endfor %}

            

                
                
            
        
    
{% endblock %}

            

              
                Copy
              
            
          

            # myapp/forms.py
from django import forms
from .models import Post

class ChangePostStatusForm(forms.Form):
    _selected_action = forms.CharField(widget=forms.HiddenInput)
    action = forms.CharField(widget=forms.HiddenInput)

    new_status = forms.ChoiceField(
        label="Novo Status",
        choices=Post.STATUS_CHOICES, # Assumindo STATUS_CHOICES definido no modelo Post
        required=True
    )
    comment = forms.CharField(
        label="Motivo/Comentário (opcional)",
        required=False,
        widget=forms.Textarea(attrs={'rows': 3})
    )

# Adicionar STATUS_CHOICES ao modelo Post
# myapp/models.py
from django.db import models

class Post(models.Model):
    STATUS_CHOICES = [
        ('draft', 'Draft'),
        ('pending', 'Pending Review'),
        ('approved', 'Approved'),
        ('rejected', 'Rejected'),
    ]
    title = models.CharField(max_length=255)
    content = models.TextField()
    status = models.CharField(max_length=20, default='draft', choices=STATUS_CHOICES)
    comment_history = models.TextField(blank=True, null=True)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.title

            

              
                Copy
              
            
          

            # myapp/admin.py (continuado)
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest, HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from .models import Post
from .forms import ChangePostStatusForm # Importar o novo formulário

@admin.register(Post)
class PostAdmin(admin.ModelAdmin):
    list_display = ('title', 'status', 'created_at')
    list_filter = ('status',)
    search_fields = ('title',)

    # Ação mark_posts_approved existente...

    def change_post_status_with_comment(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        form = None
        if 'apply' in request.POST:
            form = ChangePostStatusForm(request.POST)
            if form.is_valid():
                new_status = form.cleaned_data['new_status']
                comment = form.cleaned_data['comment']

                updated_count = 0
                for post in queryset:
                    post.status = new_status
                    if comment:
                        post.comment_history = (post.comment_history or '') + f"\n[{request.user.username}] alterado para {new_status} com comentário: {comment}"
                    post.save()
                    updated_count += 1

                self.message_user(
                    request,
                    f"{updated_count} post(s) tiveram seu status alterado para '{new_status}' e comentário adicionado.",
                    messages.SUCCESS
                )
                return HttpResponseRedirect(request.get_full_path())
        
        # Se não confirmado, ou requisição GET, mostrar o formulário de entrada
        if not form:
            form = ChangePostStatusForm(initial={
                '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
                'action': 'change_post_status_with_comment',
            })

        context = self.admin_site.each_context(request)
        context['queryset'] = queryset
        context['form'] = form
        context['action_name'] = self.change_post_status_with_comment.short_description
        context['title'] = _("Alterar Status do Post e Adicionar Comentário")

        return render(request, 'admin/change_status_action.html', context)

    change_post_status_with_comment.short_description = _("Alterar status para posts selecionados (com comentário)")

    actions = [
        mark_posts_approved,
        change_post_status_with_comment
    ]

            

              
                Copy
              
            
          

            {# templates/admin/change_status_action.html #}
{% extends "admin/base_site.html" %}
{% load i18n admin_urls static admin_modify %}

{% block extrastyle %}{{ block.super }}
    
{% endblock %}

{% block content %}
    

        

            {% csrf_token %}
            
{{ action_name }}
            
Você está prestes a alterar o status dos seguintes {{ queryset.count }} objeto(s):
            

                {% for obj in queryset %}
                    
{{ obj }}
                {% endfor %}
            
            
Por favor, selecione o novo status e opcionalmente adicione um comentário:

            {% for field in form %}
                {% if field.is_hidden %}
                    {{ field }}
                {% else %}
                    

                        {{ field.errors }}
                        {{ field.label_tag }}
                        {{ field }}
                        {% if field.help_text %}
{{ field.help_text }}
{% endif %}
                    
                {% endif %}
            {% endfor %}

            

                
                
            
        
    
{% endblock %}

            

              
                Copy
              
            
          

            # myapp/admin.py (continuado)
import requests # Você precisará instalar requests: pip install requests
# ... outras importações ...

# Assumindo ProductAdmin do exemplo anterior

class ProductAdmin(admin.ModelAdmin):
    # ... list_display, list_filter, search_fields existentes ...

    def sync_products_to_external_ims(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        # Verificar confirmação (semelhante aos exemplos anteriores, se necessário)
        if 'apply' in request.POST:
            # Simular um endpoint de API externa
            EXTERNAL_IMS_API_URL = "https://api.example.com/v1/products/sync/"
            API_KEY = "sua_chave_api_secreta" # Em um app real, use settings.py ou variáveis de ambiente

            successful_syncs = 0
            failed_syncs = []

            for product in queryset:
                data = {
                    "product_id": product.id,
                    "name": product.name,
                    "price": str(product.price), # Converter Decimal para string para JSON
                    "is_discounted": product.is_discounted,
                    # Adicionar outros dados relevantes do produto
                }
                headers = {
                    "Authorization": f"Bearer {API_KEY}",
                    "Content-Type": "application/json"
                }
                
                try:
                    response = requests.post(EXTERNAL_IMS_API_URL, json=data, headers=headers, timeout=5) # Timeout de 5 segundos
                    response.raise_for_status() # Levantar um HTTPError para respostas ruins (4xx ou 5xx)
                    successful_syncs += 1
                except requests.exceptions.RequestException as e:
                    failed_syncs.append(f"Produto {product.name} (ID: {product.id}): {e}")
                except Exception as e:
                    failed_syncs.append(f"Produto {product.name} (ID: {product.id}): Erro inesperado: {e}")

            if successful_syncs > 0:
                self.message_user(
                    request,
                    f"{successful_syncs} produto(s) sincronizados com sucesso com o IMS externo.",
                    messages.SUCCESS
                )
            if failed_syncs:
                error_message = f"Falha ao sincronizar {len(failed_syncs)} produto(s):\n" + "\n".join(failed_syncs)
                self.message_user(request, error_message, messages.ERROR)
            
            return HttpResponseRedirect(request.get_full_path())

        # Requisição GET inicial ou requisição POST não 'apply': mostrar confirmação (se desejado)
        context = self.admin_site.each_context(request)
        context['queryset'] = queryset
        context['form'] = ConfirmationForm(initial={
            '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
            'action': 'sync_products_to_external_ims',
        })
        context['action_name'] = self.sync_products_to_external_ims.short_description
        context['title'] = _("Confirmar Sincronização de Dados")

        return render(request, 'admin/confirmation_action.html', context) # Reutilizar template de confirmação

    sync_products_to_external_ims.short_description = _("Sincronizar produtos selecionados com IMS externo")

    actions = [
        # ... outras ações ...
        sync_products_to_external_ims,
    ]

            

              
                Copy
              
            
          

            # myapp/admin.py (continuado)
import csv
from django.http import HttpResponse
# ... outras importações ...

class PostAdmin(admin.ModelAdmin):
    # ... list_display, list_filter, search_fields, actions existentes ...

    def export_posts_as_csv(self, request: HttpRequest, queryset: QuerySet) -> HttpResponse:
        response = HttpResponse(content_type='text/csv')
        response['Content-Disposition'] = 'attachment; filename="posts_export.csv"'

        writer = csv.writer(response)
        # Escrever a linha de cabeçalho
        writer.writerow(['Título', 'Status', 'Criado Em', 'Prévia do Conteúdo'])

        for post in queryset:
            writer.writerow([
                post.title,
                post.get_status_display(), # Usar get_FOO_display() para campos de escolha
                post.created_at.strftime("%Y-%m-%d %H:%M:%S"),
                post.content[:100] + '...' if len(post.content) > 100 else post.content # Truncar conteúdo longo
            ])
        
        self.message_user(
            request,
            f"{queryset.count()} post(s) exportados com sucesso para CSV.",
            messages.SUCCESS
        )
        return response

    export_posts_as_csv.short_description = _("Exportar posts selecionados como CSV")

    actions = [
        # ... outras ações ...
        export_posts_as_csv,
    ]

            

              
                Copy
              
            
          

            # myproject/celery.py (configuração padrão do Celery)
import os
from celery import Celery

# Definir o módulo de configurações padrão do Django para o programa 'celery'.
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'myproject.settings')

app = Celery('myproject')

# Usar uma string aqui significa que o worker não terá que
# fazer pickle do objeto ao usar o Windows.
app.config_from_object('django.conf:settings', namespace='CELERY')

# Carregar módulos de tarefas de todas as configurações de app registradas do Django.
app.autodiscover_tasks()

@app.task(bind=True)
def debug_task(self):
    print(f'Requisição: {self.request!r}')

            

              
                Copy
              
            
          

            # myapp/tasks.py
import requests
from celery import shared_task
from django.contrib.auth import get_user_model
from django.apps import apps

@shared_task
def sync_product_to_external_ims_task(product_id, admin_user_id):
    Product = apps.get_model('myapp', 'Product')
    User = get_user_model()

    try:
        product = Product.objects.get(pk=product_id)
        admin_user = User.objects.get(pk=admin_user_id)

        EXTERNAL_IMS_API_URL = "https://api.example.com/v1/products/sync/"
        API_KEY = "sua_chave_api_secreta" # Use variáveis de ambiente ou configurações do Django

        data = {
            "product_id": product.id,
            "name": product.name,
            "price": str(product.price),
            "is_discounted": product.is_discounted,
        }
        headers = {
            "Authorization": f"Bearer {API_KEY}",
            "Content-Type": "application/json"
        }

        response = requests.post(EXTERNAL_IMS_API_URL, json=data, headers=headers, timeout=10)
        response.raise_for_status()
        
        # Registrar sucesso (por exemplo, nos logs do Django ou em um modelo específico para rastreamento)
        print(f"Produto {product.name} (ID: {product.id}) sincronizado com sucesso por {admin_user.username}.")

    except Product.DoesNotExist:
        print(f"Produto com ID {product_id} não encontrado.")
    except User.DoesNotExist:
        print(f"Usuário admin com ID {admin_user_id} não encontrado.")
    except requests.exceptions.RequestException as e:
        print(f"Sincronização da API falhou para o produto {product_id}: {e}")
    except Exception as e:
        print(f"Erro inesperado durante a sincronização para o produto {product_id}: {e}")


            

              
                Copy
              
            
          

            # myapp/admin.py (continuado)
from django.contrib import admin, messages
from django.db.models import QuerySet
from django.http import HttpRequest, HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from .models import Product # Assumindo o modelo Product do exemplo anterior
from .tasks import sync_product_to_external_ims_task # Importar sua tarefa Celery

class ProductAdmin(admin.ModelAdmin):
    # ... list_display, list_filter, search_fields existentes ...

    def async_sync_products_to_external_ims(self, request: HttpRequest, queryset: QuerySet) -> HttpResponseRedirect | None:
        if 'apply' in request.POST:
            admin_user_id = request.user.id
            for product in queryset:
                # Enfileirar a tarefa para cada produto selecionado
                sync_product_to_external_ims_task.delay(product.id, admin_user_id)
            
            self.message_user(
                request,
                f"As tarefas de sincronização de {queryset.count()} produto(s) foram enfileiradas.",
                messages.SUCCESS
            )
            return HttpResponseRedirect(request.get_full_path())

        # Requisição GET inicial ou requisição POST não 'apply': mostrar confirmação
        context = self.admin_site.each_context(request)
        context['queryset'] = queryset
        context['form'] = ConfirmationForm(initial={
            '_selected_action': request.POST.getlist(admin.ACTION_CHECKBOX_NAME),
            'action': 'async_sync_products_to_external_ims',
        })
        context['action_name'] = self.async_sync_products_to_external_ims.short_description
        context['title'] = _("Confirmar Sincronização de Dados Assíncrona")

        return render(request, 'admin/confirmation_action.html', context) # Reutilizar template de confirmação

    async_sync_products_to_external_ims.short_description = _("Enfileirar sincronização assíncrona para produtos selecionados para o IMS")

    actions = [
        # ... outras ações ...
        async_sync_products_to_external_ims,
    ]

            

              
                Copy
              
            
          

            def sensitive_action(self, request, queryset):
    if not request.user.has_perm('myapp.can_perform_sensitive_action'):
        self.message_user(request, _("Você não tem permissão para executar esta ação."), messages.ERROR)
        return HttpResponseRedirect(request.get_full_path())
    # ... lógica da ação sensível ...

            

              
                Copy
              
            
          

            # myapp/models.py
class Product(models.Model):
    # ... campos ...
    class Meta:
        permissions = [
            ("can_perform_sensitive_action", "Pode realizar ação sensível de produto"),
        ]

            

              
                Copy
              
            
          

            # myapp/admin.py
class ProductAdmin(admin.ModelAdmin):
    # ... definição de ações ...

    def get_actions(self, request: HttpRequest):
        actions = super().get_actions(request)
        # Remover a ação 'make_discounted' se o usuário não tiver uma permissão específica
        if not request.user.has_perm('myapp.change_product'): # Ou uma permissão personalizada como 'can_discount_product'
            if 'make_discounted' in actions:
                del actions['make_discounted']
        return actions

            

              
                Copy
              
            
          

            # myapp/admin_actions.py
from django.contrib import messages
from django.http import HttpRequest
from django.db.models import QuerySet

def mark_as_active(modeladmin, request: HttpRequest, queryset: QuerySet):
    updated = queryset.update(is_active=True)
    modeladmin.message_user(request, f"{updated} item(s) marcados como ativos.", messages.SUCCESS)
mark_as_active.short_description = "Marcar selecionados como ativos"

            

              
                Copy
              
            
          

            # myapp/admin.py
from django.contrib import admin
from .models import MyModel
from .admin_actions import mark_as_active

@admin.register(MyModel)
class MyModelAdmin(admin.ModelAdmin):
    list_display = ('name', 'is_active')
    actions = [mark_as_active]

            

              
                Copy
              
            
          

            # myapp/tests.py
from django.test import TestCase, Client
from django.contrib.auth import get_user_model
from django.urls import reverse
from django.contrib import admin
from .models import Product
from .admin import ProductAdmin # Importar seu ModelAdmin

User = get_user_model()

class ProductAdminActionTests(TestCase):
    def setUp(self):
        self.admin_user = User.objects.create_superuser('admin', 'admin@example.com', 'password')
        self.client = Client()
        self.client.login(username='admin', password='password')

        self.p1 = Product.objects.create(name="Product A", price=10.00, is_discounted=False)
        self.p2 = Product.objects.create(name="Product B", price=20.00, is_discounted=False)
        self.p3 = Product.objects.create(name="Product C", price=30.00, is_discounted=True)

        self.admin_site = admin.AdminSite()
        self.model_admin = ProductAdmin(Product, self.admin_site)

    def test_make_discounted_action(self):
        # Simular a seleção de produtos e a execução da ação
        change_list_url = reverse('admin:myapp_product_changelist')
        response = self.client.post(change_list_url, {
            admin.ACTION_CHECKBOX_NAME: [self.p1.pk, self.p2.pk],
            'action': 'make_discounted',
            'index': 0, # Necessário para alguma lógica interna do admin do Django
        }, follow=True)

        self.assertEqual(response.status_code, 200)
        self.assertContains(response, '2 produto(s) foram marcados como com desconto com sucesso.')
        
        self.p1.refresh_from_db()
        self.p2.refresh_from_db()
        self.p3.refresh_from_db()

        self.assertTrue(self.p1.is_discounted)
        self.assertTrue(self.p2.is_discounted)
        self.assertTrue(self.p3.is_discounted) # Este já estava com desconto

    def test_make_discounted_action_confirmation(self):
        # Para ações com confirmação, você testaria o processo em duas etapas
        change_list_url = reverse('admin:myapp_post_changelist') # Assumindo modelo Post para exemplo de confirmação
        post1 = Post.objects.create(title='Test Post 1', content='...', status='draft')
        post2 = Post.objects.create(title='Test Post 2', content='...', status='draft')

        # Etapa 1: Solicitar página de confirmação
        response = self.client.post(change_list_url, {
            admin.ACTION_CHECKBOX_NAME: [post1.pk, post2.pk],
            'action': 'mark_posts_approved',
            'index': 0,
        })
        self.assertEqual(response.status_code, 200)
        self.assertIn(b"Confirmar Ação", response.content) # Verificar se a página de confirmação foi renderizada

        # Etapa 2: Enviar formulário de confirmação
        response = self.client.post(change_list_url, {
            admin.ACTION_CHECKBOX_NAME: [post1.pk, post2.pk],
            'action': 'mark_posts_approved',
            'apply': "Sim, tenho certeza",
            'confirm': 'on', # Valor para uma caixa de seleção se renderizada como caixa de seleção
            '_selected_action': [str(post1.pk), str(post2.pk)], # Deve ser passado de volta do formulário
            'index': 0,
        }, follow=True)

        self.assertEqual(response.status_code, 200)
        self.assertContains(response, '2 post(s) foram marcados como aprovados com sucesso.')

        post1.refresh_from_db()
        post2.refresh_from_db()
        self.assertEqual(post1.status, 'approved')
        self.assertEqual(post2.status, 'approved')

            

              
                Copy